MOT_root = '/home/sean/data/MOT/2DMOT2015/'
train_file = '/home/sean/data/MOT/2DMOT2015/ImageSet/train.txt'
val_file = '/home/sean/data/MOT/2DMOT2015/ImageSet/val.txt'

train_video_list = ['train/TUD-Stadtmitte', 'train/ETH-Bahnhof', 'train/PETS09-S2L1', 'train/ADL-Rundle-6',
                    'train/KITTI-13']#, 'train/TUD-Campus', 'train/ETH-Sunnyday', 'train/ETH-Pedcross2', 'train/ADL-Rundle-8', 'train/Venice-2', 'train/KITTI-17']
train_video_len =[179, 1000, 795, 525, 261]#, 71, 354, 837, 654, 600, 145]
#kitti13_skip = [1~12,134~207]

# val_video_list = ['train/TUD-Campus', 'train/ETH-Sunnyday', 'train/ETH-Pedcross2', 'train/ADL-Rundle-8', 'train/Venice-2',
#                     'train/KITTI-17']
# val_video_len =[71, 354, 837, 654, 600, 145]

with open(train_file, 'w') as wf:
    for video, len in zip(train_video_list, train_video_len):
        for frame in range(1, len+1):
            wf.write(video+' '+str(frame).zfill(6)+'\n')

